pkill -9 -f "python .*src/train.py"
sleep 2
CLASS="keywords"
adapter_name_or_path="saves/"
EVAL_BATCH=8
LOG_FILE="${CLASS}_evaluate_$(date +%Y%m%d_%H%M%S).log"
CUDA_VISIBLE_DEVICES=0 python src/train.py \
    --stage sft \
    --model_name_or_path Qwen/Qwen2.5-7B-Instruct \
    --preprocessing_num_workers 16 \
    --finetuning_type lora \
    --quantization_method bnb \
    --template qwen \
    --flash_attn auto \
    --dataset_dir data \
    --eval_dataset ${CLASS}_eval \
    --cutoff_len 1024 \
    --max_samples 100000 \
    --per_device_eval_batch_size ${EVAL_BATCH} \
    --predict_with_generate True \
    --max_new_tokens 512 \
    --top_p 0.7 \
    --temperature 0.1 \
    --output_dir saves/eval/$CLASS \
    --trust_remote_code True \
    --do_predict True \
    --adapter_name_or_path $adapter_name_or_path
